import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

# 图例
## 第一种
x = np.linspace(0, 10, 10)
# 注：图例名称不要以下划线开头，会显示不出来
plt.plot(x, 0.5 * x, label='slow')
plt.plot(x, x, label='normal')
plt.plot(x, 2 * x, label='fast')
# 自动寻找一个最好的位置best()来放置图例
plt.legend()
# plt.plot(x, 0.5 * x, x, x, x, 2 * x)
plt.show()

## 第二种
plt.plot(x, 0.5 * x)
plt.plot(x, x)
plt.plot(x, 2 * x)
plt.legend(['slow', 'normal', 'fast'])
plt.show()

# loc参数：图里显示位置
plt.plot(x, 0.5 * x, label='slow')
plt.plot(x, x, label='normal')
plt.plot(x, 2 * x, label='fast')
plt.legend(loc='center left')
# plt.legend(loc=6)  # 同上
plt.show()

# ncol参数：控制图例中有几列
plt.plot(x, 0.5 * x, label='slow')
plt.plot(x, x, label='normal')
plt.plot(x, 2 * x, label='fast')
plt.legend(loc='center left')
plt.legend(loc=(0, 1), ncol=3)  # (0,1)是相对坐标
plt.show()
